adapt_alg="PLL"
#adapt_alg="PLL_imbalance"
dataset="cifar10"
#corruption_array=("shot_noise" "motion_blur" "snow" "pixelate" "gaussian_noise" "defocus_blur" "brightness" "fog" "zoom_blur" "frost" "glass_blur" "impulse_noise" "contrast" "jpeg_compression" "elastic_transform")
corruption="shot_noise"
test_envs=0
network="resnet18"

lr=1e-4
margin_select=0.5
margin_end=0.4
margin_descend=$(echo "scale=4; (${margin_select} - ${margin_end}) / 79" | bc)
temp_select=1
temp_loss=5
buffer_size=32
update_param='all'

seed=0
seed_array=(0)
gpu_id=0

for seed in ${seed_array[@]}
do
    CUDA_VISIBLE_DEVICES=${gpu_id} python unsupervise_adapt_cifar10c.py --corruption ${corruption} --dataset ${dataset} --adapt_alg ${adapt_alg}  --lr ${lr} --net ${network} --test_envs ${test_envs} --seed ${seed} --output tta_output/${network}/${dataset}/${corruption}/${adapt_alg} --margin_select ${margin_select} --margin_end ${margin_end} --margin_descend ${margin_descend} --temp_select ${temp_select} --temp_loss ${temp_loss} --buffer_size ${buffer_size} --update_param ${update_param}
done